import asyncio

# 用于创建函数的智能代理Agent
from llama_index.core.agent.workflow import FunctionAgent
from llama_index.llms.openai import OpenAI


def multiply(a: float, b: float) -> float:
    return a * b


# 创建一个带有计算功能的智能代理
agent = FunctionAgent(
    tools=[multiply],  # 指定代理可用的工具
    llm=OpenAI(model="gpt-4o-mini"),  # 指定使用大模型工具
    system_prompt="你是一个乐于助人的助手，可以计算两个数的乘积",  # 定义系统提示，描述代理的行为
)


# 定义异步函数
async def main():
    response = await agent.run("请帮我计算7乘以8等于多少?")
    print(str(response))


if __name__ == "__main__":
    asyncio.run(main())
